package com.example.lotterysystem.controller.handler;

import com.example.lotterysystem.common.errorcode.GlobalErrorCodeConstants;
import com.example.lotterysystem.common.exception.ControllerException;
import com.example.lotterysystem.common.exception.ServiceException;
import com.example.lotterysystem.common.result.CommonResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.rmi.ServerException;
import java.util.stream.Collectors;

/**
 * 捕获全局异常
 */
@Slf4j
@RestControllerAdvice
public class GlobalExceptionHandler {

    /**
     * 捕获 service 层抛出的异常
     * @param e
     * @return
     */
    @ExceptionHandler(value = ServerException.class)
    public CommonResult<?> serverException(ServiceException e) {
        // 打印错误日志
        log.info("ServiceException: ", e);
        // 构造异常情况下的返回结果
        return CommonResult.fail(GlobalErrorCodeConstants.INTERNAL_SERVER_ERROR.getCode(), e.getMessage());
    }

    /**
     * 捕获 controller 层抛出的异常
     * @param e
     * @return
     */
    @ExceptionHandler(value = ControllerException.class)
    public CommonResult<?> controllerException(ControllerException e) {
        // 打印错误日志
        log.info("ControllerException: ", e);
        // 构造异常情况下的返回结果
        return CommonResult.fail(GlobalErrorCodeConstants.INTERNAL_SERVER_ERROR.getCode(), e.getMessage());
    }

    /**
     * 捕获 @Valid / @Validated 注解校验异常
     * @param e
     * @return
     */
    @ExceptionHandler(value = MethodArgumentNotValidException.class)
    public CommonResult<?> validationException(MethodArgumentNotValidException e) {
        // 打印错误日志
        log.info("MethodArgumentNotValidException: ", e);
        // 获取所有字段验证错误
        String errors = e.getBindingResult()
                .getFieldErrors()
                .stream()
                .map(error -> error.getDefaultMessage())
                .collect(Collectors.joining(", "));
        // 构造异常情况下的返回结果
        return CommonResult.fail(GlobalErrorCodeConstants.BAD_REQUEST.getCode(), errors);
    }

    /**
     * 捕获未知异常
     * @param e
     * @return
     */
    @ExceptionHandler(value = Exception.class)
    public CommonResult<?> Exception(Exception e) {
        // 打印错误日志
        log.info("ControllerException: ", e);
        // 构造异常情况下的返回结果
        return CommonResult.fail(GlobalErrorCodeConstants.INTERNAL_SERVER_ERROR.getCode(), e.getMessage());
    }
}
